
import paddlehub as hub
import cv2

if __name__ == '__main__':
    module = hub.Module(name="pyramidbox_lite_mobile_mask")
    test_img_path = "C:\\Users\\tangxinmao\\Pictures\\Camera Roll\\mask.png"
    input_dict = {"data": [cv2.imread(test_img_path)]}
    results = module.face_detection(data=input_dict)
    print(results)